#!/usr/bin/env python

import numpy
from apgl.graph import VertexList as g
import apgl.graph.SparseGraph as sp
from apgl.generator.SmallWorldGenerator import SmallWorldGenerator


k = 5
p = 0.1
graph = sp.SparseGraph(g.VertexList(1000, 1))
generator = SmallWorldGenerator(p, k)
graph = generator.generate(graph)

print (graph.diameter(useWeights=True))